'''
Created on Sep 5, 2012

@author: feiwang
'''
from text_2_image.text2image import SimpleStyledText, get_default_font, multi_format_text_2_image, text_2_image
import unittest


class Test(unittest.TestCase):


    def testName(self):
        self.demo2()
    
    def demo2(self):
        image_width = 800
    
        styled_texts = []
        styled_texts.append(SimpleStyledText(
            text = u"本刊记者日前独家获悉，原重庆市公安局副局长唐建华因涉嫌受贿罪已于8月13日student is my friend被重庆市检察院批准逮捕。",
            text_color = "#0000FF",
            font = get_default_font(20),
            line_height = 24,
            image_width = image_width,
            margin=(30, 25, 15, 25),
            )
        )
        styled_texts.append(SimpleStyledText(
            text = u"9天之后的8月22日，唐建华妻子黄某亦被重庆市公安局渝北区分局刑事拘留，现押于渝北区看守所。",
            text_color = "#FF00FF",
            font = get_default_font(18),
            line_height = 24,
            image_width = image_width,
            margin=(30, 25, 15, 25),
            )
        )
        styled_texts.append(SimpleStyledText(
            text = u"接近唐建华家属的消息人士指出，唐于6月27日被调查组带走。",
            text_color = "#FF0000",
            font = get_default_font(20),
            line_height = 24,
            image_width = image_width,
            margin=(15, 25, 15, 25),
            )
        )
        styled_texts.append(SimpleStyledText(
            text = u"2010年3月1日重庆市人民政府第63次常务会议决定，任命唐建华为重庆市公安局副局长。在此之前，唐建华担任重庆市公安局经济技术开发区分局局长。",
            text_color = "#00FFFF",
            font = get_default_font(18),
            line_height = 24,
            image_width = image_width,
            margin=(15, 25, 15, 25),
            )
        )
        styled_texts.append(SimpleStyledText(
            text = u"重庆市2010年2月28日召开的“打黑除恶”专项斗争阶段总结表彰大会上，唐建华被记一等功。",
            text_color = "#FF00FF",
            font = get_default_font(20),
            line_height = 24,
            image_width = image_width,
            margin=(15, 25, 30, 25),
            )
        )
        image = multi_format_text_2_image(styled_texts)
        image.show()
        image.save("./test.png")

    def demo1(self):
        text = u'''本刊记者日前独家获悉，原重庆市公安局副局长唐建华因涉嫌受贿罪已于8月13日student is my friend被重庆市检察院批准逮捕。
        
9天之后的8月22日，唐建华妻子黄某亦被重庆市公安局渝北区分局刑事拘留，现押于渝北区看守所。

接近唐建华家属的消息人士指出，唐于6月27日被调查组带走。

2010年3月1日重庆市人民政府第63次常务会议决定，任命唐建华为重庆市公安局副局长。在此之前，唐建华担任重庆市公安局经济技术开发区分局局长。

重庆市2010年2月28日召开的“打黑除恶”专项斗争阶段总结表彰大会上，唐建华被记一等功。
'''
        
        image_width = 800
        
        ## text, image_width, font_size, line_height
        image = text_2_image(text, image_width, 20, 30)
        image.show()
        image.save("./test.jpg")

if __name__ == "__main__":
    #import sys;sys.argv = ['', 'Test.testName']
    unittest.main()